from setuptools import setup, find_packages


def readme(filelocal:str="readme.md"):
    with open(filelocal,"r",encoding='utf-8') as file:
        return file.read()



if __name__=='__main__':
    setup(
        name='deepdot-vision',
        version='0.0.1',
        description='深点视觉科技有限公司，深度学习缺陷检测模型集成与测试框架，基于pytorch框架',
        long_description=readme(),
        author='deepdot-dev',
        author_email='lishuaishuai_12@163.com',
        url='https://gitee.com/aiwebops/deepdot-vision.git',
        packages=find_packages(
            include=[
                'deepdot_vision',
                'deepdot_vision.*'
            ]
        ),

        install_requires=[
            'PyYAML',
            'torch>=1.10.0',
            'tensorboard',
        ],

        package_data={
            'deepdot_vision': [
                '*.yaml'
            ]
        }
    )
